#include <inttypes.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include "test.h"

static int scmp(const void *a, const void *b)
{
    return strcmp(*(char **)a, *(char **)b);
}

static int icmp(const void *a, const void *b)
{
    return *(int*)a - *(int*)b;
}

static int ccmp(const void *a, const void *b)
{
    return *(char*)a - *(char*)b;
}

static int cmp64(const void *a, const void *b)
{
    const uint64_t *ua = a, *ub = b;
    return *ua < *ub ? -1 : *ua != *ub;
}

/* 26 items -- even */
static const char *s[] = {
    "Bob", "Alice", "John", "Ceres",
    "Helga", "Drepper", "Emeralda", "Zoran",
    "Momo", "Frank", "Pema", "Xavier",
    "Yeva", "Gedun", "Irina", "Nono",
    "Wiener", "Vincent", "Tsering", "Karnica",
    "Lulu", "Quincy", "Osama", "Riley",
    "Ursula", "Sam"
};
static const char *s_sorted[] = {
    "Alice", "Bob", "Ceres", "Drepper",
    "Emeralda", "Frank", "Gedun", "Helga",
    "Irina", "John", "Karnica", "Lulu",
    "Momo", "Nono", "Osama", "Pema",
    "Quincy", "Riley", "Sam", "Tsering",
    "Ursula", "Vincent", "Wiener", "Xavier",
    "Yeva", "Zoran"
};

/* 23 items -- odd, prime */
static int n[] = {
    879045, 394, 99405644, 33434, 232323, 4334, 5454,
    343, 45545, 454, 324, 22, 34344, 233, 45345, 343,
    848405, 3434, 3434344, 3535, 93994, 2230404, 4334
};
static int n_sorted[] = {
    22, 233, 324, 343, 343, 394, 454, 3434,
    3535, 4334, 4334, 5454, 33434, 34344, 45345, 45545,
    93994, 232323, 848405, 879045, 2230404, 3434344, 99405644
};

static void str_test(const char **a, const char **a_sorted, int len)
{
    int i;
    qsort(a, len, sizeof *a, scmp);
    for (i=0; i<len; i++) {
        if (strcmp(a[i], a_sorted[i]) != 0) {
            t_error("string sort failed at index %d\n", i);
            t_printf("\ti\tgot\twant\n");
            for (i=0; i<len; i++)
                t_printf("\t%d\t%s\t%s\n", i, a[i], a_sorted[i]);
            break;
        }
    }
}

static void int_test(int *a, int *a_sorted, int len)
{
    int i;
    qsort(a, len, sizeof *a, icmp);
    for (i=0; i<len; i++) {
        if (a[i] != a_sorted[i]) {
            t_error("integer sort failed at index %d\n", i);
            t_printf("\ti\tgot\twant\n");
            for (i=0; i<len; i++)
                t_printf("\t%d\t%d\t%d\n", i, a[i], a_sorted[i]);
            break;
        }
    }
}

static void uint64_gen(uint64_t *p, uint64_t *p_sorted, int n)
{
    int i;
    uint64_t r = 0;
    t_randseed(n);
    for (i = 0; i < n; i++) {
        r += t_randn(20);
        p[i] = r;
    }
    memcpy(p_sorted, p, n * sizeof *p);
    t_shuffle(p, n);
}

static void uint64_test(uint64_t *a, uint64_t *a_sorted, int len)
{
    int i;
    qsort(a, len, sizeof *a, cmp64);
    for (i=0; i<len; i++) {
        if (a[i] != a_sorted[i]) {
            t_error("uint64 sort failed at index %d\n", i);
            t_printf("\ti\tgot\twant\n");
            for (i=0; i<len; i++)
                t_printf("\t%d\t%" PRIu64 "\t%" PRIu64 "\n", i, a[i], a_sorted[i]);
            break;
        }
    }
}

#define T(a, a_sorted) do { \
    char p[] = a; \
    qsort(p, sizeof p - 1, 1, ccmp); \
    if (memcmp(p, a_sorted, sizeof p) != 0) { \
        t_error("character sort failed\n"); \
        t_printf("\tgot:  \"%s\"\n", p); \
        t_printf("\twant: \"%s\"\n", a_sorted); \
    } \
} while(0)

static void char_test(void)
{
    T("", "");
    T("1", "1");
    T("11", "11");
    T("12", "12");
    T("21", "12");
    T("111", "111");
    T("211", "112");
    T("121", "112");
    T("112", "112");
    T("221", "122");
    T("212", "122");
    T("122", "122");
    T("123", "123");
    T("132", "123");
    T("213", "123");
    T("231", "123");
    T("321", "123");
    T("312", "123");
    T("1423", "1234");
    T("51342", "12345");
    T("261435", "123456");
    T("4517263", "1234567");
    T("37245618", "12345678");
    T("812436597", "123456789");
    T("987654321", "123456789");
    T("321321321", "111222333");
    T("49735862185236174", "11223344556677889");
}

int qsort_test(void)
{
    int i;

    str_test(s, s_sorted, sizeof s/sizeof*s);
    int_test(n, n_sorted, sizeof n/sizeof*n);
    char_test();
    for (i = 1023; i<=1026; i++) {
        uint64_t p[1026], p_sorted[1026];
        uint64_gen(p, p_sorted, i);
        uint64_test(p, p_sorted, i);
    }
    return t_status;
}
